import openai
import myopenaiapikey
import json
from flask import Flask

openai.api_base = myopenaiapikey.base
openai.api_key = myopenaiapikey.api


app = Flask(__name__)
#flask中对于变量的使用<>


def chat2(query):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo-0613",
        messages=[
            {"role": "assistant", 
             "content": """
                        现在你是一台基于debian的linux服务器，不需要返回任何的中文提示词，
                        只需要返回命令执行的结果，要求要做到和基于debian的linux文件系统结构一样，
                        并且能成功执行对应的命令。
                        现在你伪装的这台linux服务器ip地址是1.2.3.4，如果有提到ip相关的信息需要注意
                        你现在伪装的这台linux服务器有2个用户，分别是wwwroot，sqluser，
                        默认第一次询问的时候是使用的wwwroot身份。
                        在root文件夹里面有一个空的a.txt文件。
                        """},
            {"role": "user", "content": query}],
    )
    message = response["choices"][0]["message"]
    return message


@app.route('/admin/<cmd>')
def hello_admin(cmd):
    message = chat2(cmd)
    return message["content"]
    #return render_template
if __name__ == '__main__':

    app.run('0.0.0.0',90,debug=True)
